package exercises.leetcode;

import java.util.BitSet;

/**
 * <a href="https://leetcode-cn.com/problems/partition-equal-subset-sum/">
 * 416. 分割等和子集</a>
 *
 * <p><span color="#87CEEB">描述:</span>
 * 给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。
 * <p>
 *  
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [1,5,11,5]
 * 输出：true
 * 解释：数组可以分割成 [1, 5, 5] 和 [11] 。
 * <p>
 * 示例 2：
 * <p>
 * 输入：nums = [1,2,3,5]
 * 输出：false
 * 解释：数组不能分割成两个元素和相等的子集。
 *  
 * <p>
 * 提示：
 * <li>1 <= nums.length <= 200</li>
 * <li>1 <= nums[i] <= 100</li>
 *
 * @author or2
 * @date 2021年10月05日 时间: 11:20
 */
public class No416 {
    public boolean canPartition(int[] nums) {
        int sum = 0;
        for (int num : nums)
            sum += num;

        if ((sum & 1) == 1)
            return false;
        int target = sum >> 1;
        BitSet bitSet = new BitSet(target);

        if (nums[0] <= target)
            bitSet.set(nums[0] - 1);
        for (int i = 1, len = nums.length; i < len; i++) {
            for (int j = target - 1; j >= nums[i]; j--)
                bitSet.set(j, bitSet.get(j) || bitSet.get(j - nums[i]));
            bitSet.set(nums[i] - 1);

            if (bitSet.get(target - 1))
                return true;
        }

        return bitSet.get(target - 1);
    }
}
